import tensorflow as tf
from keras.layers import Dense
import os

if __name__ == '__main__':
    os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
    X = tf.random.normal([10, 20])

    # W = tf.Variable(tf.random.truncated_normal([20, 8], stddev=0.1))
    #
    # b = tf.Variable(tf.zeros([10,8]))
    #
    # O = X @ W + b
    # output = tf.nn.relu(O)
    # print(output)

    full_connected = Dense(units=8, activation='relu')
    output = full_connected(X)
    print(output)

    print(full_connected.kernel)
    print(full_connected.bias)
